def f x = map (* (x * x)) [0, 1, 2]

-- ==
-- tags { autodiff }
-- entry: f_jvp
-- input { 2 } output { [0, 4, 8] }
-- input { 4 } output { [0, 8, 16] }
entry f_jvp x = jvp f x 1
